#include "napi/native_api.h"
#include "common.h"
#include <iostream>
#include <fstream>
#include <stdint.h>
#include <string>

static napi_value Add(napi_env env, napi_callback_info info)
{
    size_t argc = 2;
    napi_value args[2] = {nullptr};

    napi_get_cb_info(env, info, &argc, args , nullptr, nullptr);

    napi_valuetype valuetype0;
    napi_typeof(env, args[0], &valuetype0);

    napi_valuetype valuetype1;
    napi_typeof(env, args[1], &valuetype1);

    double value0;
    napi_get_value_double(env, args[0], &value0);

    double value1;
    napi_get_value_double(env, args[1], &value1);

    napi_value sum;
    napi_create_double(env, value0 + value1 + 1, &sum);

    return sum;

}


static napi_value GenerateRandomText(napi_env env, napi_callback_info info) {

    size_t argc = 2;
    napi_value args[2] = {nullptr};

    napi_get_cb_info(env, info, &argc, args , nullptr, nullptr);

    napi_valuetype valuetype0;
    napi_typeof(env, args[0], &valuetype0);

    napi_valuetype valuetype1;
    napi_typeof(env, args[1], &valuetype1);

    // 假设传递过来的第一个参数是表示随机数种子的数字，这里获取其值（简化示例，实际可能更复杂的类型转换）
    uint32_t seed_value;
    napi_get_value_uint32(env, args[0], &seed_value);

    // 调用生成随机文本的函数
    std::string random_text = generateRandomText(seed_value);
//     std::string random_text = "你成功了!!!";
    // 将生成的C++字符串转换为napi_value类型，以便返回给ArkTS
    napi_value result;
    napi_create_string_utf8(env, random_text.c_str(), NAPI_AUTO_LENGTH, &result);

    return result;
}


static napi_value Match(napi_env env, napi_callback_info info) {

    size_t argc = 2;
    napi_value args[2] = {nullptr};

    napi_get_cb_info(env, info, &argc, args , nullptr, nullptr);

    napi_valuetype valuetype0;
    napi_typeof(env, args[0], &valuetype0);

    napi_valuetype valuetype1;
    napi_typeof(env, args[1], &valuetype1);
    
    // 口令密码
    size_t str_len;
    napi_get_value_string_utf8(env, args[0], nullptr, 0, &str_len);
    if (str_len == 0) {
        // 如果获取到的字符串长度为0，处理情况示例，可按需调整
        return nullptr;
    }
    char* buffer = new char[str_len + 1];  // 分配足够的内存空间，多分配1个字节用于存放字符串结束符'\0'
    napi_get_value_string_utf8(env, args[0], buffer, str_len + 1, nullptr);
    std::string random_text(buffer);  // 使用分配好内存并复制了内容的缓冲区构造std::string对象
    delete[] buffer;  // 释放手动分配的内存，避免内存泄漏

    // 解析音频文件
    std::string text_recog = whisper_recog();
    
    // 文本匹配
    std::string match_result = match(random_text, text_recog);
    std::string message = "语音识别结果："+text_recog+"\n文本匹配结果："+match_result;

    // 将生成的C++字符串转换为napi_value类型，以便返回给ArkTS
    napi_value result;
    napi_create_string_utf8(env, message.c_str(), NAPI_AUTO_LENGTH, &result);

    return result;
}


EXTERN_C_START
static napi_value Init(napi_env env, napi_value exports)
{
    napi_property_descriptor desc[] = {
        { "add", nullptr, Add, nullptr, nullptr, nullptr, napi_default, nullptr },
        {"generateRandomText", nullptr, GenerateRandomText, nullptr, nullptr, nullptr, napi_default, nullptr},
        {"match", nullptr, Match, nullptr, nullptr, nullptr, napi_default, nullptr}
    };
    napi_define_properties(env, exports, sizeof(desc) / sizeof(desc[0]), desc);
    return exports;
}
EXTERN_C_END

static napi_module demoModule = {
    .nm_version = 1,
    .nm_flags = 0,
    .nm_filename = nullptr,
    .nm_register_func = Init,
    .nm_modname = "entry",
    .nm_priv = ((void*)0),
    .reserved = { 0 },
};

extern "C" __attribute__((constructor)) void RegisterEntryModule(void)
{
    napi_module_register(&demoModule);
}
